# 自定义函数实现矩形序列:u[n-n1]-u[n-n2]，其中n2为学号的最后两位，
# 序列的长度为学号最末位（若末位是0， 则取10），即n2-n1=学号最末位，将结果作图画出来；
# 提示：可以利用函数np.where()
import matplotlib.pyplot as plt
import numpy as np

n_min = -25
n_max = 25
n = np.arange(n_min, n_max+1, 1)


def un(t):

    return np.where(t > 0, 1, 0)


un1 = un(n - 20)
un2 = un(n - 25)
y = un1 - un2

plt.subplot(411)
plt.stem(n, y)
plt.ylabel('x1')

plt.show()
